can you put conditions in list conprehension